home *** CD-ROM | disk | FTP | other *** search
/ Programming an RTS Game with Direct3D / Programming an RTS Game with Direct3D.iso / Examples / Chapter 14 / Example 14.1 / mapObject.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2006-07-16  |  7.9 KB  |  323 lines
  1. #include "mapObject.h"
  2. #include "unit.h"
  3. #include "building.h"
  4. #include "groupAi.h"
  5.  
  6. //Line interface to draw m_selected units/buildings with
  7. ID3DXLine *line = NULL;
  8.  
  9. //variables used for fog-of-war
  10. IDirect3DTexture9* sightTexture = NULL;
  11. ID3DXMesh *sightMesh = NULL;
  12. D3DMATERIAL9 sightMtrl;
  13. EFFECT_SELECTED *selectedEffect = NULL;
  14.  
  15. struct SIGHTVertex
  16. {
  17.     SIGHTVertex(){}
  18.     SIGHTVertex(D3DXVECTOR3 pos, D3DXVECTOR2 _uv)
  19.     {
  20.         position = pos;
  21.         uv = _uv;
  22.     }
  23.  
  24.     D3DXVECTOR3 position;
  25.     D3DXVECTOR2 uv;
  26.  
  27.     static const DWORD FVF;
  28. };
  29.  
  30. const DWORD SIGHTVertex::FVF = D3DFVF_XYZ | D3DFVF_TEX1;
  31.  
  32. void LoadMapObjectResources(IDirect3DDevice9* m_pDevice)
  33. {
  34.     D3DXCreateLine(m_pDevice, &line);
  35.  
  36.     //Sight texture
  37.     m_pDevice->CreateTexture(64, 64, 1, D3DUSAGE_DYNAMIC, D3DFMT_L8, D3DPOOL_DEFAULT, &sightTexture, NULL);
  38.  
  39.     D3DLOCKED_RECT sRect;
  40.     sightTexture->LockRect(0, &sRect, NULL, NULL);
  41.     BYTE *bytes = (BYTE*)sRect.pBits;
  42.     memset(bytes, 0, sRect.Pitch*sRect.Pitch);
  43.     
  44.     float intensity = 1.3f;
  45.     D3DXVECTOR2 center = D3DXVECTOR2(32.0f, 32.0f);
  46.     
  47.     for(int y=0;y<64;y++)
  48.         for(int x=0;x<64;x++)
  49.         {                        
  50.             float d = D3DXVec2Length(&(center - D3DXVECTOR2(x,y)));
  51.             int value = ((32.0f - d) / 32.0f) * 255.0f * intensity;
  52.             if(value < 0)value = 0;
  53.             if(value > 255)value = 255;
  54.             bytes[x + y * sRect.Pitch] = value;
  55.         }
  56.     sightTexture->UnlockRect(0);
  57.  
  58.     //D3DXSaveTextureToFile("sightTexture.bmp", D3DXIFF_BMP, sightTexture, NULL);
  59.  
  60.     //Calculate sight mesh (a simple quad)
  61.     D3DXCreateMeshFVF(2, 4, D3DXMESH_MANAGED, SIGHTVertex::FVF, m_pDevice, &sightMesh);
  62.  
  63.     //Create 4 vertices
  64.     SIGHTVertex* v = 0;
  65.     sightMesh->LockVertexBuffer(0,(void**)&v);
  66.     v[0] = SIGHTVertex(D3DXVECTOR3(-1, 0, 1),  D3DXVECTOR2(0, 0));
  67.     v[1] = SIGHTVertex(D3DXVECTOR3( 1, 0, 1),  D3DXVECTOR2(1, 0));
  68.     v[2] = SIGHTVertex(D3DXVECTOR3(-1, 0, -1), D3DXVECTOR2(0, 1));
  69.     v[3] = SIGHTVertex(D3DXVECTOR3( 1, 0, -1), D3DXVECTOR2(1, 1));
  70.     sightMesh->UnlockVertexBuffer();
  71.  
  72.     //Create 2 faces
  73.     WORD* indices = 0;
  74.     sightMesh->LockIndexBuffer(0,(void**)&indices);    
  75.     indices[0] = 0; indices[1] = 1; indices[2] = 2;
  76.     indices[3] = 1; indices[4] = 3; indices[5] = 2;
  77.     sightMesh->UnlockIndexBuffer();
  78.  
  79.     //Set Attributes for the 2 faces
  80.     DWORD *att = 0;
  81.     sightMesh->LockAttributeBuffer(0,&att);
  82.     att[0] = 0; att[1] = 0;
  83.     sightMesh->UnlockAttributeBuffer();
  84.  
  85.     //Sight MTRL
  86.     memset(&sightMtrl, 0, sizeof(D3DMATERIAL9));
  87.     sightMtrl.Diffuse = sightMtrl.Ambient = D3DXCOLOR(1.0f, 1.0f, 1.0f, 1.0f);
  88.  
  89.     //Selected Effect
  90.     selectedEffect = new EFFECT_SELECTED(m_pDevice);
  91. }
  92.  
  93. void UnloadMapObjectResources()
  94. {
  95.     if(line)line->Release();
  96.     line = NULL;
  97.  
  98.     if(sightTexture)sightTexture->Release();
  99.     sightTexture = NULL;
  100.  
  101.     if(sightMesh)sightMesh->Release();
  102.     sightMesh = NULL;
  103. }
  104.  
  105. INTPOINT GetScreenPos(D3DXVECTOR3 pos, IDirect3DDevice9* m_pDevice)
  106. {
  107.     D3DXVECTOR3 screenPos;
  108.     D3DVIEWPORT9 Viewport;
  109.     D3DXMATRIX Projection, View, World;
  110.  
  111.     m_pDevice->GetViewport(&Viewport);
  112.     m_pDevice->GetTransform(D3DTS_VIEW, &View);
  113.     m_pDevice->GetTransform(D3DTS_PROJECTION, &Projection);
  114.     D3DXMatrixIdentity(&World);
  115.     D3DXVec3Project(&screenPos, &pos, &Viewport, &Projection, &View, &World);
  116.  
  117.     return INTPOINT(screenPos.x, screenPos.y);
  118. }
  119.  
  120. //////////////////////////////////////////////////////////////////////////////////
  121. //                                MapObject                                        //
  122. //////////////////////////////////////////////////////////////////////////////////
  123.  
  124. MAPOBJECT::MAPOBJECT()
  125. {
  126.     //Sets all variables to 0, NULL or False
  127.     m_isBuilding = false;
  128.     m_pTerrain = NULL;
  129.     m_hp = m_hpMax = 0;
  130.     m_range = m_damage = 0;
  131.     m_sightRadius = 0.0f;
  132.     m_team = m_type = 0;
  133.     m_selected = m_dead = m_visible = false;
  134.     m_pTarget = NULL;
  135.     m_pDevice = NULL;
  136.     m_pGroup = NULL;
  137. }
  138.  
  139. RECT MAPOBJECT::GetMapRect(int border)
  140. {
  141.     if(border < 0)border = 0;
  142.  
  143.     RECT mr = {m_mappos.x - border, 
  144.                m_mappos.y - border,
  145.                m_mappos.x + m_mapsize.x + border,
  146.                m_mappos.y + m_mapsize.y + border};
  147.  
  148.     //Clip Rectangle to Terrain border
  149.     if(m_pTerrain != NULL)
  150.     {
  151.         if(mr.left < 0)mr.left = 0;
  152.         if(mr.right >= m_pTerrain->m_size.x)mr.right = m_pTerrain->m_size.x - 1;
  153.         if(mr.top < 0)mr.top = 0;
  154.         if(mr.bottom >= m_pTerrain->m_size.y)mr.bottom = m_pTerrain->m_size.y - 1;
  155.     }
  156.  
  157.     return mr;
  158. }
  159.  
  160. INTPOINT MAPOBJECT::GetAttackPos(INTPOINT from)
  161. {
  162.     try
  163.     {
  164.         if(m_pTerrain == NULL || !m_pTerrain->Within(m_mappos))return INTPOINT(-1, -1);
  165.  
  166.         RECT r = GetMapRect(1);
  167.         RECT mr = GetMapRect(0);
  168.  
  169.         INTPOINT bestDest(-1, -1);
  170.         float dist = 10000.0f;
  171.  
  172.         int loop = 0;
  173.  
  174.         //Find the closest available attacking position
  175.         for(int y=r.top;y<=r.bottom;y++)
  176.             for(int x=r.left;x<=r.right;x++)
  177.             {
  178.                 INTPOINT p(x, y);
  179.  
  180.                 if(!p.inRect(mr))        
  181.                 {
  182.                     MAPTILE *tile = m_pTerrain->GetTile(p);
  183.                     float d = from.Distance(p);
  184.  
  185.                     if(tile != NULL && tile->m_pMapObject == NULL && d < dist)
  186.                     {
  187.                         dist = d;
  188.                         bestDest = p;
  189.                     }
  190.                 }
  191.  
  192.                 if(loop++ > 1000)
  193.                 {
  194.                     debug.Print("Loop > 1000, MAPOBJECT::GetAttackPos()");
  195.                     debug << "R Left: " << r.left << ", Right: " << r.right << ", Top: " << r.top << ", Bottom: " << r.bottom << "\n";
  196.                     debug << "Mappos: " << m_mappos.x << ", " << m_mappos.y << "\n";
  197.                     return INTPOINT(-1, -1);
  198.                 }
  199.             }
  200.  
  201.         return bestDest;
  202.     }
  203.     catch(...)
  204.     {
  205.         return INTPOINT(-1, -1);
  206.     }
  207. }
  208.  
  209. void MAPOBJECT::PaintSelected(float time)
  210. {
  211.     if(m_isBuilding)
  212.     {
  213.         BUILDING *build = (BUILDING*)this;        
  214.         if(build->m_training)
  215.             build->m_pTrainingEffect->Render();
  216.     }
  217.  
  218.     if(!m_selected || !selectedEffect)return;
  219.  
  220.     if(!m_isBuilding)
  221.     {
  222.         selectedEffect->Set(m_hp / (float)m_hpMax, 
  223.                             m_position + D3DXVECTOR3(0.0f, 0.2f, 0.0f),
  224.                             time, 
  225.                             1.2f);
  226.     }
  227.     else
  228.     {
  229.         selectedEffect->Set(m_hp / (float)m_hpMax, 
  230.                             m_position + D3DXVECTOR3(0.0f, 1.2f, 0.0f),
  231.                             time, 
  232.                             m_mapsize.x + m_mapsize.y + 2.0f);
  233.     }
  234.  
  235.     selectedEffect->Render();
  236. }
  237.  
  238. void MAPOBJECT::RenderSightMesh()
  239. {
  240.     if(m_pDevice == NULL || sightTexture == NULL || sightMesh == NULL)return;
  241.  
  242.     //Set world transformation matrix
  243.     D3DXMATRIX world, pos, sca;
  244.  
  245.     //Position the mesh at the center of the map object
  246.     D3DXMatrixTranslation(&pos, m_position.x, m_position.y, m_position.z);
  247.  
  248.     //Scale the mesh to the sight radius of the mapobject (XZ plane)
  249.     D3DXMatrixScaling(&sca, m_sightRadius, 1.0f, m_sightRadius);
  250.  
  251.     D3DXMatrixMultiply(&world, &sca, &pos);
  252.     m_pDevice->SetTransform(D3DTS_WORLD, &world);
  253.  
  254.     //Set texture and material
  255.     m_pDevice->SetTexture(0, sightTexture);
  256.     m_pDevice->SetMaterial(&sightMtrl);
  257.  
  258.     //Draw the sight mesh
  259.     sightMesh->DrawSubset(0);
  260. }
  261.  
  262. std::vector<MAPOBJECT*> MAPOBJECT::GetTargetsWithinRange(int theRange)
  263. {
  264.     std::vector<MAPOBJECT*> enemies;
  265.  
  266.     try
  267.     {
  268.         if(m_pTerrain == NULL)return enemies;
  269.  
  270.         RECT r = GetMapRect(theRange);
  271.  
  272.         for(int y=r.top;y<=r.bottom;y++)
  273.             for(int x=r.left;x<=r.right;x++)
  274.             {
  275.                 MAPTILE *tile = m_pTerrain->GetTile(x, y);
  276.  
  277.                 if(tile != NULL && tile->m_pMapObject != NULL)
  278.                     if(tile->m_pMapObject->m_team != m_team && !tile->m_pMapObject->m_dead)
  279.                         if(m_mappos.Distance(INTPOINT(x, y)) <= theRange || m_isBuilding)
  280.                             enemies.push_back(tile->m_pMapObject);
  281.             }
  282.     }
  283.     catch(...)
  284.     {
  285.         debug.Print("Error in MAPOBJECT::GetTargetsWithinRange()");
  286.     }
  287.  
  288.     return enemies;
  289. }
  290.  
  291. MAPOBJECT* MAPOBJECT::BestTargetToAttack(std::vector<MAPOBJECT*> &enemies)
  292. {
  293.     if(enemies.empty())return NULL;
  294.  
  295.     int lowestCost = 10000;
  296.     MAPOBJECT *bestTarget = NULL;
  297.  
  298.     try
  299.     {
  300.         for(int i=0;i<enemies.size();i++)
  301.             if(enemies[i] != NULL && !enemies[i]->m_dead)
  302.             {
  303.                 INTPOINT attackPos = enemies[i]->GetAttackPos(m_mappos);
  304.  
  305.                 if(m_pTerrain->Within(attackPos) && m_pTerrain->PositionAccessible(this, attackPos))
  306.                 {
  307.                     int cost = enemies[i]->m_hp + m_mappos.Distance(enemies[i]->m_mappos) * 30;
  308.  
  309.                     if(cost < lowestCost)
  310.                     {
  311.                         bestTarget = enemies[i];
  312.                         lowestCost = cost;
  313.                     }
  314.                 }
  315.             }
  316.     }
  317.     catch(...)
  318.     {
  319.         debug.Print("Error in MAPOBJECT::BestTargetToAttack()");
  320.     }
  321.  
  322.     return bestTarget;
  323. }